package com.atguigu.gmall.common.config;

import com.alibaba.fastjson.JSON;
import com.rabbitmq.client.ConfirmCallback;
import com.rabbitmq.client.Return;
import com.rabbitmq.client.ReturnCallback;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.io.IOException;

@Component
public class MQProducerAckConfig implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnCallback {

    // 注入rabbit模板
    @Autowired
    private RabbitTemplate rabbitTemplate;

    // 将rabbit模板和我们设置的确认模式绑定起来
    // 这个注解表示在类初始化的时候就加载下面的init方法，init方法是自己写的，可以随便起名字
    @PostConstruct
    public void init(){
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * 这个方法是每次都会执行的，检测是否往交换机中发送消息成功，如果成功，ack则是true，失败false
     * @param correlationData 消息主体
     * @param ack  是否发送成功
     * @param cause 成功失败的消息体
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        if (ack) {
            System.out.println("消息发送成功：" + JSON.toJSONString(correlationData));
        } else {
            System.out.println("消息发送失败：" + cause + " 数据：" + JSON.toJSONString(correlationData));
        }

    }

    /**
     * 这个方法如果发送成功的话是不执行的，如果消息没有往队列中发送成功，则会执行本方法
     * @param message 消息主体
     * @param replyCode 状态码
     * @param replyText 状态消息
     * @param exchange 交换机
     * @param routingKey 路由key
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        System.out.println("message = " + message);
        System.out.println("replyCode = " + replyCode);
        System.out.println("replyText = " + replyText);
        System.out.println("exchange = " + exchange);
        System.out.println("routingKey = " + routingKey);
    }
}
